Armstrong number program in python